mock a function within a function